#! /bin/bash

source activate yolo && cd /root/code/mmyolo
mim train mmyolo /root/code/mmyolo/configs/yolov5/yolov5_s-v61_syncbn_fast_8xb16-300e_coco.py \
    --amp \
    --cfg-options train_dataloader.batch_size=128 \
    val_dataloader.batch_size=128 \
    train_dataloader.num_workers=16 \
    val_dataloader.num_workers=16